import os
import shutil

def opt2D_to_Raw_data():


    count1=0
    # print("是否进行了  %5 and %10 文件的生成，输入yes or no" )
    # string8=input()
    # if string8=='yes':
    root_dir = './Data'
    # 指定根目录路径
    # root_dir = './finally_par_result'
    # 指定目标文件夹路径
    # target_folder = 'C:\\Users\\admin\\Desktop\\Data_process\\Raw_data'
    target_folder = '.\\Data_process\\Raw_data'
    # 你想要查找的文件名关键词
    keyword = 'opt2D'

    # 确保目标文件夹存在
    if not os.path.exists(target_folder):
        os.makedirs(target_folder)

    # 遍历根目录及其所有子目录
    for root, dirs, files in os.walk(root_dir):
        # 检查当前目录的深度，如果大于3，则停止遍历
        if root[len(root_dir):].count(os.sep) > 2:
            del dirs[:]  # 清除dirs列表，避免继续遍历更深层次的目录
        else:
            # 遍历当前目录下的所有文件
            for file in files:
                # 检查文件名是否包含关键词'opt2D'
                if keyword in file:
                    # 构建源文件的完整路径
                    source_file_path = os.path.join(root, file)
                    # 构建目标文件的完整路径
                    target_file_path = os.path.join(target_folder, file)
                    # 复制文件
                    shutil.copy2(source_file_path, target_file_path)
                    count1=count1+1
    # print(f'文件迁移数为：{count1}')
    return count1
                    # print(f"文件 {file} 已从 {root} 复制到 {target_folder}")
    # else:
    #     root_dir = './two_par_result'
    # # 指定根目录路径
    # # root_dir = './finally_par_result'
    # # 指定目标文件夹路径
    #     target_folder = 'C:\\Users\\admin\\Desktop\\Data process\\Raw_data'
    #     # 你想要查找的文件名关键词
    #     keyword = 'opt2D'

    #     # 确保目标文件夹存在
    #     if not os.path.exists(target_folder):
    #         os.makedirs(target_folder)

    #     # 遍历根目录及其所有子目录
    #     for root, dirs, files in os.walk(root_dir):
    #         # 检查当前目录的深度，如果大于3，则停止遍历
    #         if root[len(root_dir):].count(os.sep) > 2:
    #             del dirs[:]  # 清除dirs列表，避免继续遍历更深层次的目录
    #         else:
    #             # 遍历当前目录下的所有文件
    #             for file in files:
    #                 # 检查文件名是否包含关键词'opt2D'
    #                 if keyword in file:
    #                     # 构建源文件的完整路径
    #                     source_file_path = os.path.join(root, file)
    #                     # 构建目标文件的完整路径
    #                     target_file_path = os.path.join(target_folder, file)
    #                     # 复制文件
    #                     shutil.copy2(source_file_path, target_file_path)
    #                     print(f"文件 {file} 已从 {root} 复制到 {target_folder}")


# opt2D_to_Raw_data()